knn python from scratch